package com.hqyj.javaSpringBoot.modules.account.service.impl;

import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import javax.transaction.Transactional;

import com.hqyj.javaSpringBoot.util.RedisUtils;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.hqyj.javaSpringBoot.modules.account.dao.UserDao;
import com.hqyj.javaSpringBoot.modules.account.dao.UserRoleDao;
import com.hqyj.javaSpringBoot.modules.account.entity.User;
import com.hqyj.javaSpringBoot.modules.account.entity.UserRole;
import com.hqyj.javaSpringBoot.modules.account.service.UserService;
import com.hqyj.javaSpringBoot.modules.common.entity.Result;
import com.hqyj.javaSpringBoot.modules.common.entity.SearchBean;
import com.hqyj.javaSpringBoot.util.MD5Util;
/**
 * @Description：UserServiceImpl
 * @Author ：Cwd
 * @Date ：2021/8/26 11:37
 */
@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserDao userDao;
    @Autowired
    private UserRoleDao userRoleDao;
    @Autowired
    private RedisUtils redisUtils;

    @Override
    public User getUserByUserNameAndPassword(String userName, String password) {
        return userDao.getUserByUserNameAndPassword(userName, MD5Util.getMD5(password));
    }

    @Override
    public User getUserByUserName(String userName) {
        List<User> users = Optional
                .ofNullable(userDao.getUserByUserName(userName))
                .orElse(Collections.emptyList());
        return users.isEmpty() ? null : users.get(0);
    }

    @Override
    public Result<User> login(User user) {
        int maxNum = 5;
        // 从数据库获取 错误登录次数
        String key = String.format("failed_login_count_%s", user.getUserName());
        int loginFailCount = redisUtils.get(key) == null ? 0 : (Integer)redisUtils.get(key);
        // 如果错误登录次数 > max，直接返回错误信息
        if (loginFailCount >= maxNum) {
            return new Result<User>(Result.ResultCode.FAILED.code,
                    String.format("该账户连续登录错误 %s 次，账户锁定 30 s.", loginFailCount));
        }
        // 获得 subject
        Subject subject = SecurityUtils.getSubject();
        // 包装令牌
        UsernamePasswordToken token = new UsernamePasswordToken(
                user.getUserName(),MD5Util.getMD5(user.getPassword()));
        token.setRememberMe(user.getRememberMe());

        try {
            // 调用 subject login 方法，进行身份验证
            subject.login(token);
            // 调用 subjectCheck roles 方法，进行资源授权
            subject.checkRoles();
            // 处理 session
            subject.getSession().setAttribute("user", subject.getPrincipal());
            // 登录成功，将错误次数归 0
            redisUtils.set(key, 0);
        }catch (Exception e){
            // Redis 的登录错误次数 + 1
            loginFailCount += 1;
            redisUtils.set(key, loginFailCount);
            // 如果错误登录次数 + 1  = max ，Redis 设置该key的过期时间
            if (loginFailCount < maxNum) {
                return new Result<User>(Result.ResultCode.FAILED.code,
                        String.format("登录错误，还剩 %s 次", (maxNum - loginFailCount)));
            } else {
                // 返回错误消息
                redisUtils.expire(key, 15);
                return new Result<User>(Result.ResultCode.FAILED.code,
                        String.format("该账户连续登录错误 %s 次，账户锁定 30 s.", loginFailCount));
            }
//            return new Result<User>(Result.ResultCode.FAILED.code, e.getMessage());
        }
        // 返回 result
        return new Result<User>(Result.ResultCode.SUCCESS.code,
                "login success.",(User)subject.getPrincipal());

        /*User temp = userDao.getUserByUserNameAndPassword(user.getUserName(),
                MD5Util.getMD5(user.getPassword()));
        if (temp != null) {
            return new Result<User>(Result.ResultCode.SUCCESS.code, "Success...", temp);
        } else {
            return new Result<User>(Result.ResultCode.FAILED.code, "Failed...");
        }*/

    }

    @Override
    public void logout() {
        Subject subject = SecurityUtils.getSubject();
        subject.logout();
    }

    @Override
    @Transactional
    public Result<User> insertUser(User user) {
        List<User> users = Optional
                .ofNullable(userDao.getUserByUserNameOrEmail(user.getEmail(), user.getUserName()))
                .orElse(Collections.emptyList());
        if (users.size() > 0) {
            return new Result<User>(Result.ResultCode.FAILED.code, "User Name or email is repeat.");
        }

        user.setCreateDate(LocalDateTime.now());
        user.setPassword(MD5Util.getMD5(user.getPassword()));
        userDao.insertUser(user);
        if (user.getRoles() != null) {
            user.getRoles().stream()
                    .forEach(item -> {userRoleDao.insertUserRole(new UserRole(user.getId(), item.getId()));});
        }
        return new Result<User>(Result.ResultCode.SUCCESS.code, "Insert success", user);
    }

    @Override
    @Transactional
    public Result<User> updateUser(User user) {
        List<User> users = Optional
                .ofNullable(userDao.getUserByUserNameOrEmail(user.getEmail(), user.getUserName()))
                .orElse(Collections.emptyList());
        if (users.stream().filter(item -> item.getId() != user.getId()).count() > 0) {
            return new Result<User>(Result.ResultCode.FAILED.code, "User Name or email is repeat.");
        }

        userDao.updateUser(user);
        if (user.getRoles() != null && !user.getRoles().isEmpty()) {
            userRoleDao.deleteUserRoleByUserId(user.getId());
            user.getRoles().stream()
                    .forEach(item -> {userRoleDao.insertUserRole(new UserRole(user.getId(), item.getId()));});
        }

        return new Result<User>(Result.ResultCode.SUCCESS.code, "Update success", user);
    }

    @Override
    public User getUserById(int id) {
        return userDao.getUserById(id);
    }

    @Override
    @Transactional
    public Result<Object> deleteUserById(int id) {
        userDao.deleteUserById(id);
        userRoleDao.deleteUserRoleByUserId(id);
        return new Result<Object>(Result.ResultCode.SUCCESS.code, "Delete success");
    }

    @Override
    public PageInfo<User> getUsersBySearchBean(SearchBean searchBean) {
        searchBean.initSearchBean();
        PageHelper.startPage(searchBean.getCurrentPage(), searchBean.getPageSize());
        return new PageInfo<User>(Optional
                .ofNullable(userDao.getUsersBySearchVo(searchBean))
                .orElse(Collections.emptyList()));
    }

    @Override
    public List<User> getUsersByKeyWord(String keyWord) {
        SearchBean searchBean = new SearchBean();
        searchBean.setKeyword(keyWord);
        searchBean.initSearchBean();
        return Optional.ofNullable(userDao.getUsersBySearchVo(searchBean)).orElse(Collections.emptyList());
    }
}
